iT邦幫忙

2022 iThome 鐵人賽

DAY 9
0

我們將昨天影像增強的功能掛在是人是馬的範例上再來一次,首先我們來看before,我們可以看到在多次的epoch後,訓練的accuracy可以非常接近1而loss非常接近0,但測試圖庫的結果如同昨天未增強的情況,accuracy接近持平但loss增長更嚴重:
https://ithelp.ithome.com.tw/upload/images/20220913/20141158cm0NLyQioc.png
https://ithelp.ithome.com.tw/upload/images/20220913/20141158oT8B8RcZaT.png

然後我們加上以下的增強設定:

datagen_aug = ImageDataGenerator(
      rescale=1./255,
      rotation_range=40,
      width_shift_range=0.2,
      height_shift_range=0.2,
      shear_range=0.2,
      zoom_range=0.2,
      horizontal_flip=True,
      fill_mode='nearest')

我們將訓練和測試的圖庫都增強,結果如下:
https://ithelp.ithome.com.tw/upload/images/20220913/20141158TJnO7XPZem.png
https://ithelp.ithome.com.tw/upload/images/20220913/20141158TmCmOcMf4K.png

可以看到反而accuracy在約只有0.65的附近跳動。而如果只將測試的圖庫增強,結果是相似的:
https://ithelp.ithome.com.tw/upload/images/20220913/20141158tD6SGPbfyL.png
https://ithelp.ithome.com.tw/upload/images/20220913/20141158ArrZrnWFoy.png

這結果課堂的說明是:

The image augmentation introduces a random element to the training images. But if the validation set doesn't have the same randomness, then its results can fluctuate like this.

測試我特別挑選的真實圖庫,首先是原本沒有增強的:
https://ithelp.ithome.com.tw/upload/images/20220913/20141158qC9sMdyds7.png

然後是訓練的圖庫有增強的,結果都判斷為人:
https://ithelp.ithome.com.tw/upload/images/20220913/20141158kddiGdb0Zp.png

由於這訓練時間就相對長,所以就只有iterate 2次的樣本數,但如上次提到的每次訓練的結果並不相同,以目前這個結果來說並沒有肯定的結論。


上一篇
Day07 新手村的練習課程-先轉一轉,翻一翻,移一移,再
下一篇
Day9 新手村的練習課程-站在巨人的肩膀上
系列文
來創造一個AI角色吧-新手的探尋之路30
圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

尚未有邦友留言

立即登入留言